Python Convert multiple line code to a single line code